#!/bin/bash
set -e

echo "开始安装Docker..."

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# 检查权限
if [[ $EUID -ne 0 ]]; then
    log_error "请使用root权限或sudo运行此脚本"
    exit 1
fi

# 更新软件包索引
log_info "更新软件包列表..."
apt-get update

# 安装依赖
log_info "安装依赖包..."
apt-get install -y \
    apt-transport-https \
    ca-certificates \
    curl \
    gnupg \
    lsb-release

# 添加Docker的官方GPG密钥
log_info "添加Docker的官方GPG密钥..."
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg

# 添加稳定版仓库
log_info "添加Docker仓库..."
echo \
  "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu \
  $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null

# 更新软件包索引（再次）
apt-get update

# 安装Docker引擎
log_info "安装Docker引擎..."
apt-get install -y docker-ce docker-ce-cli containerd.io

# 启动Docker服务并设置开机启动
log_info "启动Docker服务..."
systemctl enable docker
systemctl start docker

# 将当前用户加入docker组（如果知道当前用户名）
if [ -n "$SUDO_USER" ]; then
    log_info "将用户 $SUDO_USER 添加到docker组..."
    usermod -aG docker "$SUDO_USER"
else
    log_warn "无法确定当前用户，请手动将用户添加到docker组："
    echo "  usermod -aG docker <用户名>"
fi

# 验证安装
log_info "验证安装..."
docker --version
if [ $? -eq 0 ]; then
    log_info "Docker安装成功！"
else
    log_error "Docker安装失败"
    exit 1
fi

log_info "请重新登录以使组权限生效，或执行 'newgrp docker'"
EOF